package cn.xeblog.plugin.tools.read.util;

import cn.xeblog.plugin.tools.read.entity.Book;
import cn.xeblog.plugin.tools.read.entity.LegadoBook;

import javax.swing.table.DefaultTableModel;
import java.util.List;

/**
 * 表格数据工具类
 *
 * @author LYF
 * @date 2022-07-19
 */
public class TableDataUtil {

    /**
     * 将书籍转换为DefaultTableModel
     *
     * @param books 书籍列表
     * @return 转换后的DefaultTableModel
     */
    public static DefaultTableModel bookToTableModel(List<Book> books) {

        DefaultTableModel model = new DefaultTableModel();
        // 添加列数据
        model.setColumnIdentifiers(new String[]{"书名", "作者", "类型"});
        // 添加行数据
        for (Book book : books) {
            String[] row = new String[]{book.getName(), book.getAuthor(), book.getType().getName()};
            model.addRow(row);
        }
        return model;
    }

    /**
     * 将legado书籍转换为DefaultTableModel
     *
     * @param books 书籍列表
     * @return 转换后的DefaultTableModel
     */
    public static DefaultTableModel legadoBookToTableModel(List<LegadoBook> books) {
        DefaultTableModel model = new DefaultTableModel();
        // 添加列数据
        model.setColumnIdentifiers(new String[]{"书名", "作者"});
        // 添加行数据
        for (LegadoBook book : books) {
            String[] row = new String[]{book.getName(), book.getAuthor()};
            model.addRow(row);
        }
        return model;
    }
}
